package com.at.bigdata.spark.core.rdd.operator.transform

import org.apache.spark.rdd.RDD
import org.apache.spark.{HashPartitioner, SparkConf, SparkContext}

/**
 *
 * @author cdhuangchao3
 * @date 2023/3/19 9:52 PM
 */
object Spark14_RDD_partitionBy {

  def main(args: Array[String]): Unit = {
    // 环境准备
    val sparkConf = new SparkConf()
      .setMaster("local[*]")
      .setAppName("Operator")
    val sc = new SparkContext(sparkConf)

    // TODO 算子 - key - value类型
    val rdd1 = sc.makeRDD(List(1, 2, 3, 4), 2)

    val mapRDD: RDD[(Int, Int)] = rdd1.map((_, 1))
    // RDD => PairRDDFunctions
    // 隐式转换（二次编译）
    // 根据指定的分区规则，对数据进行重分区
    val newRDD = mapRDD.partitionBy(new HashPartitioner(2))
//      .saveAsTextFile("output")
    newRDD.partitionBy(new HashPartitioner(2))
      .saveAsTextFile("output")
  }

}
